import os


def search_files(root_dir):
    """
    在指定的根目录下搜索特定类型的文件。

    :param root_dir: 起始目录
    :param file_type: 要搜索的文件类型（如 '.txt'）
    :return: 包含找到的文件路径的列表
    """
    depth = 2 # 遍历深度
    file_type = '.xml'  # 你要搜索的文件类型

    found_files = []

    # 检查深度是否为正整数
    if depth < 0:
        raise ValueError("深度必须是非负整数")

    # 起始目录的绝对路径深入
    root_depth = len(root_dir.split(os.path.sep))
    # 遍历目录
    for dirpath, dirnames, filenames in os.walk(root_dir):
        # 带上盘符的路径长度
        dir_depth = len(dirpath.split(os.path.sep))
        if dir_depth - root_depth <= depth:
            for filename in filenames:
                if filename.endswith(file_type):
                    found_files.append(os.path.join(dirpath, filename))

    return found_files


# 使用示例
root_dir = r'E:\H.恒康信息科技\B.版本相关\F.升级记录\Z.中台相关\J.二代脚本\V2.46\V2.46_表结构'  # 替换为你的根目录路径


matching_files = search_files(root_dir)
print(matching_files)  # 打印找到的文件路径
